BatchToSpaceND

将输入张量的 batch 维度按 block 因子分解并重排到空间维度，随后根据 crops 参数对输出空间范围进行裁剪。

  • 输入形状： [batch, height, width, channel]

  • block_size： [block_h, block_w]

  • crops： [top, bottom, left, right]

输入：

  • input - 输入数据地址。

  • input_shape - 输入形状，格式为 [batch, height, width, channel]。

  • block_size - 分块因子，格式为 [block_h, block_w]。

  • crops - 裁剪参数，格式为 [top, bottom, left, right]。

  • core_mask(int, 可选) - 核掩码（仅适用于共享存储版本）。

输出：

  • output - 输出数据地址。

支持平台：

FT78NE MT7004

备注

  • FT78NE 支持的数据类型： fp32、fp64、cplx64、cplx128、int16、int8、int32。

  • MT7004 支持的数据类型： fp32、fp16、cplx64、int16、int32。

共享存储版本:

void i8_batch_to_space_nd_s(int8_t *input, int8_t *output, const int *input_shape, const int *block_size, const int *crops, int core_mask)
void i16_batch_to_space_nd_s(int16_t *input, int16_t *output, const int *input_shape, const int *block_size, const int *crops, int core_mask)
void i32_batch_to_space_nd_s(int32_t *input, int32_t *output, const int *input_shape, const int *block_size, const int *crops, int core_mask)
void hp_batch_to_space_nd_s(float16 *input, float16 *output, const int *input_shape, const int *block_size, const int *crops, int core_mask)
void fp_batch_to_space_nd_s(float *input, float *output, const int *input_shape, const int *block_size, const int *crops, int core_mask)
void dp_batch_to_space_nd_s(double *input, double *output, const int *input_shape, const int *block_size, const int *crops, int core_mask)
void c64_batch_to_space_nd_s(float *input, float *output, const int *input_shape, const int *block_size, const int *crops, int core_mask)
void c128_batch_to_space_nd_s(double *input, double *output, const int *input_shape, const int *block_size, const int *crops, int core_mask)

C 调用示例：

 1// MT7004 示例（共享存储多核，DDR 地址）
 2#include <stdio.h>
 3
 4int main(int argc, char *argv[]) {
 5    float *input = (float *)0x81000000;
 6    float *output = (float *)0x82000000;
 7    int input_shape[4] = {400, 2, 2, 3};
 8    int block_size[2] = {2, 2};
 9    int crops[4] = {0, 0, 0, 0};
10    int core_mask = 0xff;
11    fp_batch_to_space_nd_s(input, output, input_shape, block_size, crops, core_mask);
12    return 0;
13}

私有存储版本:

void i8_batch_to_space_nd_p(int8_t *input, int8_t *output, const int *input_shape, const int *block_size, const int *crops)
void i16_batch_to_space_nd_p(int16_t *input, int16_t *output, const int *input_shape, const int *block_size, const int *crops)
void i32_batch_to_space_nd_p(int32_t *input, int32_t *output, const int *input_shape, const int *block_size, const int *crops)
void hp_batch_to_space_nd_p(float16 *input, float16 *output, const int *input_shape, const int *block_size, const int *crops)
void fp_batch_to_space_nd_p(float *input, float *output, const int *input_shape, const int *block_size, const int *crops)
void dp_batch_to_space_nd_p(double *input, double *output, const int *input_shape, const int *block_size, const int *crops)
void c64_batch_to_space_nd_p(float *input, float *output, const int *input_shape, const int *block_size, const int *crops)
void c128_batch_to_space_nd_p(double *input, double *output, const int *input_shape, const int *block_size, const int *crops)

C 调用示例：

 1// MT7004 示例（私有存储单核，AM 地址）
 2#include <stdio.h>
 3
 4int main(int argc, char *argv[]) {
 5    float *input = (float *)0x10000000;
 6    float *output = (float *)0x10020000;
 7    int input_shape[4] = {400, 2, 2, 3};
 8    int block_size[2] = {2, 2};
 9    int crops[4] = {0, 0, 0, 0};
10    fp_batch_to_space_nd_p(input, output, input_shape, block_size, crops);
11    return 0;
12}